package com.zhuang.reggie.filter;

import com.alibaba.fastjson.JSON;
import com.zhuang.reggie.common.BaseContext;
import com.zhuang.reggie.common.R;
import lombok.extern.slf4j.Slf4j;
import org.springframework.util.AntPathMatcher;

import javax.servlet.*;
import javax.servlet.annotation.WebFilter;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;

/**
 * 自定义登录检查的过滤器组件
 **/
@Slf4j
@WebFilter(filterName = "loginCheckFilter", urlPatterns = "/*")//拦截所有请求路径(单个*是servlet的写法)
public class LoginCheckFilter implements Filter {

    //路径匹配器，支持通配符 /**
    public static final AntPathMatcher PATH_MATCHER = new AntPathMatcher();

    /**
     * 过滤方法
     */
    @Override
    public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException {
        HttpServletRequest httpServletRequest = (HttpServletRequest) request;
        HttpServletResponse httpServletResponse = (HttpServletResponse) response;

        //1、获取本次请求的URI
        String requestURI = httpServletRequest.getRequestURI();

        //log.info("拦截到的请求：{}", requestURI);

        //定义不需要处理的请求路径
        String[] urls = new String[]{
                "/employee/login",//这里放行的是登录请求
                "/employee/logout",//这里放行的是退出请求
                "/backend/**",//这里放行的是所有静态资源，包括页面(但是不包括页面里面的请求)
                "/front/**", //这里放行的是所有静态资源，包括页面(但是不包括页面里面的请求)
                "/common/**", //这里放行的是文件上传下载请求
                "/user/sendMsg", //移动端发送短信
                "/user/login",//移动端登录
                "/doc.html",//knife4j生成的静态资源
                "/webjars/**",
                "/swagger-resources",
                "/v2/api-docs"
        };
        //2、判断本次请求是否需要处理
        boolean check = check(urls, requestURI);

        //3.如果不需要处理，则直接放行
        if (check) {
            //log.info("不需要处理的请求：{}", requestURI);
            chain.doFilter(httpServletRequest, httpServletResponse);
            //放行后，后面的代码也不需要执行了，直接return结束
            return;
        }

        //log.info("要处理的请求：{}", requestURI);

        //4-1、判断登录状态(后台)，如果已登录，则直接放行
        if (httpServletRequest.getSession().getAttribute("employee") != null) {
            //log.info("用户已登录，用户id为：{}", httpServletRequest.getSession().getAttribute("employee"));
            //调用BaseContext设置当前登录用户的id
            Long empId = (Long) httpServletRequest.getSession().getAttribute("employee");
            BaseContext.setCurrentId(empId);
            chain.doFilter(httpServletRequest, httpServletResponse);
            return;
        }

        //4-2、判断登录状态(移动端)，如果已登录，则直接放行
        if(httpServletRequest.getSession().getAttribute("user") != null){
            //log.info("用户已登录，用户id为：{}",httpServletRequest.getSession().getAttribute("user"));

            Long userId = (Long) httpServletRequest.getSession().getAttribute("user");
            BaseContext.setCurrentId(userId);

            chain.doFilter(request,response);
            return;
        }

        //5、如果未登录则返回未登录结果，通过输出流方式向客户端页面响应数据(前端页面有一个响应拦截器接收该数据)
        //将R对象转为Json格式响应给前端
        httpServletResponse.getWriter().write(JSON.toJSONString(R.error("NOTLOGIN")));
        return;
    }

    /**
     * 路径匹配，检查本次请求是否需要放行
     *
     * @param urls
     * @param requestURI
     * @return
     */
    public boolean check(String[] urls, String requestURI) {
        for (String url : urls) {
            //调用匹配器的匹配方法; 支持通配符 /**
            boolean match = PATH_MATCHER.match(url, requestURI);
            //返回true表示匹配上了
            if (match){
                //匹配上了直接返回true无需再循环
                return true;
            }
        }
        //循环走完了还没有匹配上返回false
        return false;
    }
}
